package com.xxxx.car.appointment.config;

import cn.hutool.json.JSONUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

@ServerEndpoint(value = "/imserverSingle")
@Component
@Slf4j
public class WebSocketSingleServer implements InitializingBean {

    /**
     * 记录当前在线连接数
     */
    public static final Map<String, Session> sessionMap = new ConcurrentHashMap<>();

    // @Resource
    // ImSingleService imSingleService;

    // static ImSingleService staticImSingleService;

    /**
     * 连接建立成功调用的方法
     * @param session
     */
    @OnOpen
    public void onOpen(Session session){
        sessionMap.put(session.getId(),session);
        log.info("[OnOpen] 新建连接,session={} , 当前在线人数为: {}",session.getId(),sessionMap.size());
    }

    /**
     * 连接关闭调用的方法
     * @param session
     */
    @OnClose
    public void onClose(Session session){
        sessionMap.remove(session.getId());
        log.info("[OnOpen] 有一连接关闭,session={} , 当前在线人数为: {}",session.getId(),sessionMap.size());
    }

    /**
     * 收到客户端消息后调用的方法
     * 后台收到客户端发送过来的消息
     * onMessage 是一个消息的中转站
     * 接受 浏览器 socket.send 发送过来的 json数据
     * @param message
     */
    @OnMessage
    public void onMessage(String message,Session fromSession){
        log.info("服务端收到消息:{}",message);
       // ImSingle imSingle = JSONUtil.toBean(message,ImSingle.class);
       // imSingle.setTime(new Date());
        // 存储数据到数据库
        // staticImSingleService.add(imSingle);
        // String json = JSONUtil.toJsonStr(imSingle);
        // this.sendAllMessage(json);
        // log.info("[onMessage] 发送消息: {}",json);
    }

    @OnError
    public void onError(Session session,Throwable error){
        log.error("[onError] 发送错误",error);
    }




    /**
     * 服务器发送消息给所有客户端
     * @param message
     */
    private void sendAllMessage(String message){
        try {
            for(Session session : sessionMap.values()){
                log.info("服务器给客户端[{}]发送消息{}",session.getId(),message);
                session.getBasicRemote().sendText(message);
            }
        }catch (Exception e){
            log.error("服务器发送消息给客户端失败",e);
        }
    }


    @Override
    public void afterPropertiesSet() throws Exception {

    }
}
